package com.answer.aop;

import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;
import org.springframework.util.StopWatch;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.UUID;

/**
 * 请求响应日志 AOP
 **/
@Aspect
@Component
@Slf4j
public class LogInterceptor {

    /**
     * 拦截器方法，用于在方法执行前后进行计时，并记录请求和响应的日志
     *
     * @param point 切入点对象，用于获取目标方法的参数等信息
     * @return 目标方法的执行结果
     * @throws Throwable 如果目标方法抛出异常，则进行传递
     */
    @Around("execution(* com.answer.controller.*.*(..))")
    public Object doInterceptor(ProceedingJoinPoint point) throws Throwable {
        // 创建停表对象，用于计算方法执行时间
        StopWatch stopWatch = new StopWatch();
        // 开始计时
        stopWatch.start();
        // 获取当前请求属性
        RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
        // 从请求属性中获取HttpServletRequest对象
        HttpServletRequest httpServletRequest = ((ServletRequestAttributes) requestAttributes).getRequest();
        // 生成请求唯一标识符
        String requestId = UUID.randomUUID().toString();
        // 获取请求路径
        String url = httpServletRequest.getRequestURI();
        // 获取请求参数
        Object[] args = point.getArgs();
        // 将请求参数转换为字符串形式
        String reqParam = "[" + StringUtils.join(args, ", ") + "]";
        // 记录请求开始的日志，包括请求ID、路径、客户端IP和请求参数
        log.info("request start，id: {}, path: {}, ip: {}, params: {}", requestId, url,
                httpServletRequest.getRemoteHost(), reqParam);
        // 执行目标方法
        Object result = point.proceed();
        // 记录请求结束的日志，包括请求ID和总执行时间
        stopWatch.stop();
        long totalTimeMillis = stopWatch.getTotalTimeMillis();
        log.info("request end, id: {}, cost: {}ms", requestId, totalTimeMillis);
        // 返回目标方法的执行结果
        return result;
    }

}

